# -*- coding: utf-8 -*-
from selenium import webdriver
from selenium.webdriver.common.by import By
import os

# 初始化浏览器（以 Chrome 为例）
driver = webdriver.Chrome()
driver.get("http://10.59.9.8:7007/boss/login#jsgl")  # 替换为你的目标网页

# 定位元素（示例：通过 ID 定位）
element = driver.find_element(By.CSS_SELECTOR, "form div ul li:nth-child(3) span img")
# 确保截图目录存在
screenshot_dir = "screenshots"
os.makedirs(screenshot_dir, exist_ok=True)

# 直接截取元素并保存
element.screenshot(os.path.join(screenshot_dir, "element.png"))

print("元素截图已保存！")
driver.quit()
